Draft Less, Retrieve More: Hybrid Tree Construction for Speculative Decoding
少草稿,多检索:投机解码的混合树构建
🧭 投机解码 · 草稿树优化
Zhejiang University · Qwen Applications Business Group of Alibaba · University of Science and Technology of China
通过剪枝释放预算并用检索补偿,构建混合草稿树,打破稠密树与剪枝树的帕累托权衡
📌 工作参考价值(4 点)
- 推理加速的预算重新分配:将剪枝释放的 VRAM 与算力用于检索高预测性 token,适合电商搜索/推荐系统中低延迟生成场景的资源调度。
- 无损、免训练的即插即用:方法无需额外训练,可直接嵌入现有投机解码框架,对模型效果零损失,便于在 Agent 对话或推荐结果生成等在线服务中快速验证。
- 检索与生成的深度融合:通过拓扑嫁接将检索 token 无缝插入树结构,启发在生成式推荐中,利用外部知识库或高频序列库补充生成草稿,提升命中率。
- 长上下文与大规模模型的稳定增益:在 235B 大模型和 64K 长序列上表现突出,为电商大模型(如商品描述生成、多轮对话)的实时推理提供高效加速方案。